Changing class name to match file name