Egy kis refactorálás