Small edits based on suggestions
Minor style edits