Fixed up the shaders - hopefully :S