Add controle for diff command