Fix some yardoc issues

This PR fixes some yardoc issues I encountered when generating the docs