Coursera Data Science Track - Data Scientist's Toolbox Week 3 assignment