Добавил очистку + косметические изменения в коде