Небольшие изменения в коде контроллера.