Some unittests finally, yay!