Snyggat till koden till mainmenu