updates to the cli
fixed the bug