Incorporating own code into manimlib

I would like to add functions to the `manimlib` that are callable in any new scene class.

Therefore, I created some new classes (e.g. `class_name`) with respective functions, in `function.py` and added a map to the `manimlib` folder, say `functions`. The I added a line to the `imports.py` file found in the manimlib folder:

`from manimlib.functions.function import *
`

Now when I start a new Scene, I can indeed reach those new functions, when I type

`class_name.function_name(parameters)
`
but Visual Code does not seem to recognize those. That is, they run and they work just as they should, but Visual Studio does not auto-complete or suggest the functions or parameters. Is this solvable? Because this now takes a lot of going back and forth and the whole idea to automatize this is lost.