Adding z_projection to Figure script