Fix typo and pep8
Use relative imports