Change of class name in the comments