Added a main class