added feature extraction correctly