Add test and docs