Updated class name in comments