My work for the data science class