Assignment 2 - update testSolution