Update gremlin_query.py