Adding new features to our model