Writing the test solution