Data Science Coding Interview