Added main class