Not discard self loops?

Not really an issue but curious how we can get the algorithm to work with self loops? I see the code filters them out but if we need to preserve self loops, how do we go about it? I gave it a shot by skipping the filtering altogether but the code then fails in other areas where it's computing forces etc. Any pointers?