added test class for CLI