I'm not sure about moving the position of `ax`.
I don't worry too much about backwards compatibility in this case but we should have a consistent pattern across plots.