Mechanical Calculation Python