Add initial version of problem 2 code.