Data Science Toolkit assignment 1