Added feature extraction module