- Update README
- More comments in code