Update comments following API update