Fixed problems with YARD docs