Finalize problem set 2