Fixing Issues with the NetworkModel. Though the performance still