Data science assignement