Put code back to production run function