Ändrade lite i koden