Microsoft Data Scientist Python Question