Clean up plot.go
Make degree of Tree configurable