Added support for main class in build