modified infra script