Improve YARD documentation