Update documentation and comments