Change set/get network state implementation. Change the NN result for agent. Add tests for set/get network state.