Fixes #1, add a token param