Moved Shader Code

I needs cleaning up. But works for now.