Add support to custom uniforms