Optimizing CUDA kernel
Whitespace fix