Update graph_builder.py

revert old change and add new connection