updated comment and unit test