Changed class name in code