Class Assignment for Data Science Toolbox