Snyggat till koden, lagt till lite fler kommentarer.