Adding wallace tree stuff