Updated shader

Shader now supports depth