Kommenterat klasserna och rensat upp lite