added menu screen, end screen, and commented code