Data Science Toolkit Assignment