revise feature extraction