Fixed so that the tests satisfies the demos for sprint 2

Gjorde så att dom fungerar, lade inte ner mycket tid på att vi använder exakt rätt metoder just nu eftersom vissa saker inte är implemnterade. Men lämnade TODOs på lite skitsaker.