Added tests in the feature spec