add comments

- add comments
- reorganize