Accidentally removed main class definition. Re-added.