added a small test and updated docs