Adding run_all support