Improve typings

[braposo]

Now that we've fully transitioned to TS (#23) there's still some work to be done to have a better type definition in the library.

This is a process that takes some time so it will be part of the post 2.0 release tasks.