Cli tests

- Added tests for the cli
- Updated the cli file for argument parsing

Fixes #130