Небольшая оптимизация кода и исправление оишбок.