Skrivit lite kommentarer och städat koden lite