Added TODO to shader.rs