Included different graph types in method graph_builder