Data Science Course Task