Allow pruning of root node and fix colours