Adding main class to build.gradle