Update functions based on API return changes