Add test and yardoc