Montolivo set for Milan switch