Add problem set 2 solutions