Montolivo set for Milan return