added imports, use unittest