Testing the code