Finished cli_ui and modify test.py