Modified py program