my assignment for the first section on the Data Scientist course