Change class name to match file name