Using version 2.0.1. In Python, using a function return type causes the following lines to not color properly. See attached screenshot. **Screenshots** With return type:  Without return type:  Thanks!