Added some basic unit tests.

Lots more to come