прописать можно, но договор оспорим.
Чтоб квартира осталась у супруга, надо сперва на него оформить долю др супруга и только затем прописать условия в договоре. Иначе нет смысла.