Add unit tests

I get it, this is the default right?