Flyttat funktionalitet och rensat upp lite smått