Update polynomial_features_memory.py