Simple stable marriage sample with Python