Finished week3 assignment