Update documentation based on review