Would like to contribute updating the type interface

Here is an example of what I'm thinking:

<img width="219" alt="image" src="https://user-images.githubusercontent.com/40436363/161335340-b0434556-d93f-4b36-8c11-c66e70f50bcd.png">
