added main menu, game over screen and other fixes