Data Science assignment task