Krunic 'training with Milan'