beginning to add some unit tests