update method signature for #20 and #22
Merge pull request #23 from ucsf-ckm/revise-api

update method signature for #20 and #22