Add support for tox