add code to generate axis_labels dict