For Data Scientist's Toolkit Coursera assignment