Add cli tests