added some quick unit tests