Add all models to graph creation