Update docstrings in attention allocation dynamics tool