Add test for cli