correct the code