Adds ability to run tests