Separate main method to support multiple runtime without dependency
Create its own class for YARN runtime