New context now also working for CUDA