modified shuaipython/main.py