Update API functions for new API