Working on feature extraction