Refine some todos