Update docs, add test