Small fix of the net_surgery.ipynb

Fixed the error in the last block - not imported matplotlib and numpy