Update docs according to #16

Reflecting recent API change.