Completed problem set 4