minor code styling edits