Fixed layout bug in shader. Now also working on NVIDIA hardware.