Just some style edits