Koden verkar fungera nu. Dags att optimera.