Update normalize_features.py