[What] Refine the code