Changed import statements (python3 doesn't allow relative imports anymore)