Updated arguments for cuda kernel