Added parameter to BFS