Gjort noen endringer for å få litt mer clean code