change feature extraction as Model member function