Add type hints

[boorlakov]

Type hints enhance code readability and maintainability by specifying the types of variables, function parameters, and return values. Adding type hints to the project will improve code quality and facilitate better understanding for developers working on the project.