// added some unit tests