public ArrayList<Integer> getNeighbours(int node, int size) {
		
		ArrayList<Integer> neighbours = new ArrayList<Integer>();
		
		if(node == 0) { //Top-left
			neighbours.add(node+1);
			neighbours.add(node+size);
		}
		else if (node == size-1) { //Top-right
			neighbours.add(node-1);
			neighbours.add(node+size);
		}
		else if (node == size*size-size) { //Bottom-left
			neighbours.add(node+1);
			neighbours.add(node-size);
		}
		else if (node == size*size-1) { //Bottom-right
			neighbours.add(node-1);
			neighbours.add(node-size);
		}
		else if (node%size == 0) { //Left
			neighbours.add(node+1);
			neighbours.add(node+size);
			neighbours.add(node-size);
		}
		else if ((node+1)%size == 0) { //Right
			neighbours.add(node-1);
			neighbours.add(node+size);
			neighbours.add(node-size);
		}
		else if (node<size) { //Top
			neighbours.add(node+1);
			neighbours.add(node-1);
			neighbours.add(node+size);
		}
		else if (node>(size*size-size)) { //Bottom
			neighbours.add(node+1);
			neighbours.add(node-1);
			neighbours.add(node-size);
		}
		else { //Centre (all 4 neighbours)
			neighbours.add(node+1);
			neighbours.add(node-1);
			neighbours.add(node+size);
			neighbours.add(node-size);
		}
		
		return neighbours;
	}