move set_output() in main loop