Pong implementation in Python + pygame