Small changes to flow of program.
Added to the game loop.