Update Watts_Strogatz.py