Adding Unit test