Add assignment 2 solution