Finished Problem set 2 for lesson 1