Add tests
Add tests
Fix cli.js