Ändrat om koden lite i lab 5 uppgift 2