Move import statement on top for PEP8 compliancy.