Problem Set 1: Bouncing Ball in Python