Isolated button class & moved game loop to main