Finished problem set 2a