refactored: Улучшен код