Adding Unit Test cases