It would be useful to have type-hints as it allows better reasoning about the code and avoiding some errors