99% klar. Behöver snygga till koden och kommentarer