Add Main class for test.