Add types throughout

[m-bossart]

Most of the code does not have types explicitly included. Needed to make the code more performant.