Added Feature Extraction Code