Update docs + tests