Update API usage

Updated API usage