Added ability to run chosen tests