Added tests to working solution.