Updated documentation for reflect changes from review