Add test run mode