whoops. i need to save the feature selector as well