changed class names to Main