minor code restyling