code with the required changes