Removed unnecessary python import