Add subsections to problem 2