print a bit more
clean up a little bit
add graphs to ipython notebook