Flashdata Messages no Codeigniter

//Controller
<?php
public function excluir($cliente_id){
        $this->load->model('cliente_model');
        $delete = $this->cliente_model->delete($cliente_id);
        if($delete){
            $this->session->set_flashdata('sucesso', 'Cliente excluído com sucesso.');
        }
        else{
            $this->session->set_flashdata('erro', 'Não foi possível excluir o cliente.');
        }
        redirect('cliente');
    }
?>

//View
<?php
        if($this->session->flashdata('erro')){
            echo '<div class="erro">'.$this->session->flashdata('erro').'</div>';
        }
        if($this->session->flashdata('sucesso')){
            echo '<div class="sucesso">'.$this->session->flashdata('sucesso').'</div>';
        }
?>