PEP8
Remove unnecessary import