Add some unit tests

This is a start at least.. I have some catching up to do!