DataSci assignment