Rewording of the comment and adding a new test case for the feature added