assignment for data science course wk4