Minor changes in the geometry shader