refactor pong a litte