YARD documentation updated