you could add few type hints to the code files so that it would be easy to interpret the code in a much easier way. #3

This can be taken as a suggestion and not as a serious issue, if needed any help with the type hints part then i can help you with that part.