Data science course assignment