Renamed import
Flake8
Merge pull request #5 from lextoumbourou/master

Flake8 and fixed a thing