Adding solution to assignment 2
Adding example code to verify results of assignment 2