finishing unit tests