Use CLI args for configuration
Refactor into a class