starting to write some unit tests