Change class name to match the file name