Add ability to run tests