ENH: Plot now accepts ax=None