Reformat code correctly