Dovizioso working on balance of the bike