Allow annotation of models with 3 arguments
Updated TreePotential for use with Any edge indices