Works on sample data
Switched to BFS