My solution to the problem set