ADD: adding in the plotting function
FIX: reverted to plotting the distribution
ADD: bfs visited with tests
FIX: bugs