Refactoring

Об’єднав два пулл-реквеста в один