Updated the discussion section