Add comment to tell to don't use the old neighbourhoods.py version and point to the new ones.

Also make sure the input is on the CPU, so a transfer is added when needed.