Add tests
Do UnitTest mother fuckaaaaaaaaa