Added class names to main() function