Remove unused import
Add tests for argument parsing