A side comment for the future - according to pep8 it's better to group imports by the namespace/source.