add the tree diagram to the test, assert instructions are correct in the test