Made imports compatible with Py3