Changes

int i = blockIdx.y * blockDim.y + threadIdx.y;

float total = c*dt / dx;

if (i < nx && j < nx)

{