Test doc improvements