Update watts_strogatz.py