Adding a test and some doc