Added tox support so it's easier to run tests locally