Add feature extraction method