halfway there on the demo
getting through demo correctly now
merging in featurization demo