ändrat lite i koden