Refine the init script