Fixed ball.py and main.py