Add unit test for note