Create count_neighbours.py