Add more doc improvements