Allow main class to be set via command line