Added test cases for Unit Testing