Started adding CUDA support