Improve example figure (axis labels etc.)