Implementing Unit test