Update comments for consistency