Stable Marriage Problem Python