Move some imports to module level