Updated to compile for CUDA... stills not working :(