BFS in Python