Update BFS code