[python mode] Also highlight the cls variable

Closes #2275