Some modifications in assignment2, network part.