ENH: modified:   output.py