IntroToDataScience, exercise lesson 4