Add in some unit tests