DataScience assignment