Небольшая ревизия кода.