Add unit test for feature