Comments and add the better code for BFS