Немного изминений в коде :-)