Data Science Interview Basic Question