minor code edits.