Rename `plot_context` to `axis`