add feature extraction code