Python Challenge: 3rd questions solution