added test class for main