Small edits

Just some minor changes