Python fod data analysis - chapter 2 - example