Separated feature extraction step from data loading