Update correction.py
Merge pull request #111 from blha303/patch-5

okay