Add TypeScript types

[itsezc]

For better DX, it'd be great if we can add TypeScript support