Python style import line

Fixes #3