reformat code, a small modify.