Exercise for Data Science Course