updates to main.cu
fixes and comments