ax is now in the wrong position in docstring, given that it's again the last explicit keyword