Made some style edits