Adding tests to the feature