Add argparse arguments to run script
Fix numerical instability, clean up some run.jl args