Data Science Toolkit class assignment