network.py seems to be working on training set