Added main class now it is working