Rich's function now also gives axis as output