Finish problem set 2