Modified code.