Done. By the way, the axis argument is now checked ahead and brought back between `-ndim` and `ndim - 1` if needed. If the argument is invalid, a IndexError is raised.