Change graph_model to use neo4j 2 labels