specify if you want to save figure or only the axis