make sure argument to sqrt is positive
fix: set titles on ax instead of fig
rename qutip.graph to qutip.visualization since it now contains a growing number of different visualization functions