Attempting To Fix Python Generation