DatSci assignement