Balotelli back in Milan training