Еще чуть-чуть тестирования

Слегка причесал логику тестирования