'Bonus' python beginner program