Update implementation to reflect API changes