Fixed Python 3 import problems