Simple BMI calculator in Python