Updated documentation and code comments