Data Science assignemt