Update documentation based on code review