Pass axis to plotting functions. Optional grid.