Data Science Toolkit assignment