remove_y_axis function

Remove y-axis or parts of it