зачем поправлять?
ваша логика прозрачна.

нет задачи впарить.