Bakayoko 'wants Milan return'