Add codeboxes for problem 2