simplied code in preparation for adding offset argument
simplied code in preparation for adding offset argument

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