add test logic