Change tf2_code to args.tf2_code in run