adding feature extraction