Problem set 3 done, first try.