bug: certain lines of code in Python-only levels don't highlight

[evemartin]

Describe the bug
Some lines of code in Python-only levels don't highlight, specifically ones to do with incrementing variables and ones that call procedures.

To Reproduce
Steps to reproduce the behaviour:

1. Go to a Python-only level that involves incrementation, i.e. Python Den Level 16

2. Click "Solve" to get the model solution, then run the model solution

3. Notice that only the lines of code involving my_van highlight, not the while loop or the other lines controlling count

4. Go to a Python-only level that involves a procedure, i.e. Python Den Level 57

5. Click "Solve" to get the model solution, then run the model solution

6. Notice that the lines of code where the procedure is called don't highlight - once again only the lines of code involving my_van highlight

Expected behaviour
Every line of code should highlight while it's running, including the ones to do with incrementing variables or procedures

Screenshots

Private Zenhub Video

Private Zenhub Video