snyggat upp koden