Improve documentation across all files