- Update code according to API changes