Minor code reformatting and comment