Allegri: 'I'd like to coach Balotelli'