Minor code reformatting