working on pong.py