clean up correct.py