Alles verbeterd, alleen unit test nog fixen