Add output file support