Update snapshots_and_feature_extraction.py