Pong like game in pygame