revise the code