Finishing the game loop assignment