Update src/cannonballgame.py

My first try to clean up someone elses code.