Tweaks to features.ipynb