Added a the main class