Adding one test for unit testswq
Implementing unit tests and correcting the mock object