More improvements on the docs