GPU update - clean a little the kernel