Add YARD comments

Fixes #2. Generated docs locally and they look good.