Rewrite to the code