Started working on main class