AC Milan head coach Massimiliano Allegri believes Mario Balotelli could become a world-class player if he continues to impress on his return to Serie A.