Minor edits to BFS and added TODOs