Modify the API

Modify API work