Python For Data Science Assignment 4