Fixed the warning : SyntaxWarning: import * only allowed at module level