Add test output mode