Add CUDA
Add CUDA functionality. Should probably go in new file correlation_CUDA.py.