update CLI code