Added plug magnitudes to find_neighbors.py