Updated main class and pong class