Uppdaterat lite mer kod