Could be a nice improvement. Easier to maintain then the current type-hints in doc-string

https://docs.python.org/3/library/typing.html