Update main.cu