Add unit test for diff