Improve documentation
Add comments
Add configuration parameter