Working on his fitness