Add support for geometry shader