fix: set titles on ax instead of fig
fix: set titles on ax instead of fig

cgranade/qutip,qutip/qutip,zasdfgbnm/qutip,cgranade/qutip,zasdfgbnm/qutip,anubhavvardhan/qutip,anubhavvardhan/qutip,qutip/qutip