Add new docs and fix broken test