lite justering på main och gamestate för att få det att fungera