Tiny cleanups to demos/CollisionBalls.py