update function signature
Merge pull request #1 from janl/master

Update function signature.