Adding unit test file