Improvements to FactorGraphViewer