Update drop_ball_height.py