this is for data science cousera homework