Changes to feature extraction script