Ändrat struktur på koden