added initial code for feature extraction