Update comments per review