BFS, DFS added
Minor changes