add arg to run()