Add test and sample usage