Improve documentation and code clarity.