Fix arguments
Merge pull request #4 from geoffrey-brier/fix-arguments

Fix arguments