Small changes to graph. Hopefull will fix problems.