Added test, updated docs