Solution to Running Time

Modified correctness-invariant.py