Minor change to learn_features' function definition