Update small_graph.py

Just added some comments