Update featurizer.py