Change classify to classification and add clarification in docstring