Add the option to overwrite the base window class