Simplified logic
Results for the test data