Add tests for run()