move graph_db to class and rename file