added new unit tests