refactor - drobna uprava kodu