Add comments on main class