Add unit test file