Montella set to sign extension