Coursera Data Analysis - Quiz 2 on python