modification of feature extraction module