Update docs based on feedback