Add MainClass for testing