updated network.py to take in a more general argument