Alla testar avklarade, ska piffa upp koden lite och försöka implementera felhantering.