Data Science Assignment