Pong written in python