Small code reformat