Might be just as well doing ```ax = plt.axes()``` and using that as an argument to ```rainbow_text```.