minor edits

models/simple_atmosphere.py

@@ -3,26 +3,30 @@ import numpy as np
 
 # ATMOSPHERE MODEL:
 
+
 def T_air(h):
-    if h >= 0 and h <= 11000:
+    if 0 <= h <= 11000:
         res = 288.15 - 0.0065 * h
-        return res
-    elif h > 11000 and h <= 20000:
+    elif 11000 < h <= 20000:
         res = 216.65
-        return res
     elif h >= 20000:
         res = 216.65 + 0.0010 * (h - 20000)
-        return res
+    return res
+
+
 def p_air(h):
-    if h >= 0 and h <= 11000:
+    if 0 <= h <= 11000:
         res = 101325 * ((288.15 - 0.0065 * h)/288.15) ** 5.25577
-        return res
-    elif h > 11000 and h <= 20000:
+    elif 11000 < h <= 20000:
         res = 22632 * np.exp(-(h - 11000)/6341.62)
-        return res
     elif h > 20000:
         res = 5474.87 * ((216.65 + 0.0010 * (h - 20000))/216.65) ** (-34.163)
-        return res
+    return res
+
+
 def rho_air(h):
     res = p_air(h)/(R_air * T_air(h))
-    return res
+    return res
+
+
+def