Ping pong game in Python