Small edits made in code