Add more docs
Test fixes