Fixed docs
Small test improvements