Refactor shaders