Add more comments + fix typos
Add more comments + fix typos