finished states.Menu and fixed main.Main