Updated doc

Updated the doc after some feedback.