Data Science - Assignment