Writing Unit test.