Added tests to solution.