Update the docs based on feedback