Made imports compatible with Python 2 and 3.4