data science code