Basic data science internship question