Update wei_following_network.py