Improve run-tests script