made chages to feature extraction