Add tests for cli