Add YARD docs