Add args and update the docs and readme.