added test logic