UI changes

Added main menu, game over screen, splash screen, removed unnecessary
things in comments, and also restructured program.