Add YARD for documentation
Some changes. ?