Revise code: