Cleaned up the menu a bit

Also, added the game over -> menu. Not fancy, but it works.