Please correct the code