Changed class name to mode