Final versions of problem set 5.1