The Data Scientist's Toolbox Week 1 assignment