Allow updater to define the main class