enhancement: coloring custom defined functions

[baggiponte]

Would it be possible to add coloring for custom defined functions, to highlight the items underlined in the screen attached? Or this depends on JupyterLab/the lsp? Thanks!