Update calc_area.py
diff --git a/calc_area.py b/calc_area.py
index 4d4050a6c2..6a51627439 100644
--- a/calc_area.py
+++ b/calc_area.py
@@ -1,19 +1,42 @@
 # Author: PrajaktaSathe
 # Program to calculate the area of - square, rectangle, circle, and triangle -
-shape = int(input("Enter 1 for square, 2 for rectangle, 3 for circle, or 4 for triangle: "))
-if shape == 1:
-  side = float(input("Enter length of side: "))
-  print("Area of square = " + str(side**2))
-elif shape == 2:
-  l = float(input("Enter length: "))
-  b = float(input("Enter breadth: "))
-  print("Area of rectangle = " + str(l*b))
-elif shape == 3:
-  r = float(input("Enter radius: "))
-  print("Area of circle = " + str(3.14*r*r))
-elif shape == 4:
-  base = float(input("Enter base: "))
-  h = float(input("Enter height: "))
-  print("Area of rectangle = " + str(0.5*base*h))
-else:
-  print("You have selected wrong choice.")
\ No newline at end of file
+import math as m
+
+def main():
+    shape = int(input("Enter 1 for square, 2 for rectangle, 3 for circle, 4 for triangle, 5 for cylinder, 6 for cone, or 7 for sphere: "))
+    if shape == 1:
+      side = float(input("Enter length of side: "))
+      print("Area of square = " + str(side**2))
+    elif shape == 2:
+      l = float(input("Enter length: "))
+      b = float(input("Enter breadth: "))
+      print("Area of rectangle = " + str(l*b))
+    elif shape == 3:
+      r = float(input("Enter radius: "))
+      print("Area of circle = " + str(m.pi*r*r))
+    elif shape == 4:
+      base = float(input("Enter base: "))
+      h = float(input("Enter height: "))
+      print("Area of rectangle = " + str(0.5*base*h))
+    elif shape == 5:
+        r = float(input("Enter radius: "))
+        h = float(input("Enter height: "))
+        print("Area of cylinder = " + str(m.pow(r, 2)*h*m.pi))
+    elif shape == 6:
+        r = float(input("Enter radius: "))
+        h = float(input("Enter height: "))
+        print("Area of cone = " + str(m.pow(r, 2)*h*1/3*m.pi))
+    elif shape == 7:
+        r = float(input("Enter radius: "))
+        print("Area of sphere = " + str(m.pow(r, 3)*4/3*m.pi))
+    else:
+      print("You have selected wrong choice.")
+
+    restart = input("Would you like to calculate the area of another object? Y/N : ")
+
+    if restart.lower().startswith("y"):
+        main()
+    elif restart.lower().startswith("n"):
+        quit()
+
+main()

Adds 3 more (3d) shapes, instead of just multiplying by 3.14 I changed it to pi, and I also added an option to run the program again.