Modify Function In Python