Make library type-checkable

[iron3oxide]

Minor refactor as I came across some issues while using this library and having pyright enabled. Hope I understood the spirit of the code correctly and thus made acceptable changes. If so, a new pypi release would be great! Thanks in advance