Re-edit the code