Minor - reformat code