Update for API changes (Fixes #84)