Add basic test for CLI