Доработка директивы return