Небольшая доработка :)