Minor style edits for arxiv version