Fixed relative imports
Now supports Python3 too