code of data science project