Finishing unit tests