add docs + fix tests