Small code reformatting