A simple pong game [Python]