Add python wrapper to infect file