remove_y_axis function

Remove y axis or parts of it