Fixed YARD docs bugs