Небольшие изменения, съэкономил одну переменную