Update NetworkModule.py