edit cli.py