Fix bug, add cli