Support for highlighting code by different color according to types like in VSCode

[YongcaiHuang]

Elevator Pitch

any plan to support the highlight code by different color according to types (instance, function, class, package etc.) like in VSCode?

Motivation

the highlight code by different color according to types can quick help me to find and identify what I have write, especially when there are many lines and content in one cell.

in VSCode:

in Jupyter Notebook:

by comparing the same code snippet, higher readability in VSCode, even when I change different way of defining. In Jupyter Notebook, function or class is only highlighted following dot .. Also, the colors for package, class, function, instance are different with higher readability.