Update api.py, add missing module