Небольшие изменения и фикс ошибки