Added marshalling to the models for graph traversal