Uploaded for the public use of second assignment in the data science course