make easier to change n_features