YARD doc updates