Add cli.py
Fix cli