fix improved code to run args