Fixed up visualization functions

Now that I look at it, the network seems to have a strange weighting
bug. That's currently the biggest problem I can see.