Python code for a game of pong