Add more typescript annotations

**Is your feature request related to a problem? Please describe.**
The code base is not fully type checked, which could lead to errors and longer development times.

**Describe the solution you'd like**
Change all files from .js/.jsx to .ts/.tsx, and add more type annotations.

**Additional context**
There is already some typescript support, but it should be expanded on.