minor edits on code style