add feature extraction