Readability improvements on variable names, type hints, doc strings, etc (#1)

* readability improvements on variable names and type hints

* add .DS_Store to gitignore

---------

Co-authored-by: Isay Katsman <isaykatsman@google.com>