Update bfs.py