Add flag support for corrections.py