Modified the main class in the pom file