Minor code restyling