main: Change the type argument of use_shader_program() to gl_shader_stage.

This allows it to be called from a loop.