Update cuda_main.cpp