Code edits to create commented solution