Add basic argument parser and a few test functions