fix 0-axis selection and add comments to the example