Fixed Problem with shader

Some other problem still needs to be finished.