Small changes to CUDA addition