slight work on main class