Adjust the main logic code

[maomincoding]

801

Make corresponding adjustments for this point.

[krausest]

Thanks. Here's the comparison (on the left the updated keyed version, on the right the version from the previous run): As expected a bit slower.