- Added a few comments
- Cleaned up code