Adding some edits to the discussion section.