Initial Unit test

Still very basic, but should be extended further.