Adding a class that can extract features from the Data