adding some unit tests