Minor style edits to code