fix PEP8 and some errors in feature_selection in the prototype