fikset en bug, og ryddet litt i koden