Data Science Python Code