Cleaned shader

Made the shader more readable