Kommentert koden og lagt til noen funksjoner