Pong written in python and pygame