pong code snippets in python