Städat upp koden lite