Modified code to be tested