Added main class and easy code