improvements made to feature_extraction.py