Improve tox configuration