optimize api.py