Solution to problem set 2