fixing class name in cli.py