Небольшой рефакторинг кода и fix #12