Expand functionality of set_shader callback function