Fixing an error and adding test_args