Support 3d simulations

main.py

@@ -1,5 +1,6 @@
 #!./venv/bin/python
 import argparse
+import typing
 
 import data
 import physics
@@ -8,6 +9,7 @@ import physics
 class Args:
     filename: str
     gravity: float
+    dimensions: typing.Literal[2, 3]
 
 
 if __name__ == "__main__":
@@ -27,11 +29,18 @@ if __name__ == "__main__":
         type=float,
         default=1.
     )
+    parser.add_argument(
+        "-d",
+        "--dimensions",
+        type=int,
+        choices=[2, 3],
+        default=2
+    )
 
     args: Args = parser.parse_args()
 
     physics.G = args.gravity
 
-    objects = data.parse_csv(args.filename)
+    objects = data.parse_csv(args.filename, dimensions=args.dimensions)
     a = data.Animator(*objects)
     a.show()