- Add reset functionality
- Add comments