Rewrite the code.