Add the homework 2 solution