Update comment blocks and documentation