Improved code comments and README.md