Improve README and code comments