Update

- clean code
- add comment