Update test to 2.0

And fix documentation function