Create watts_strogatz.py