diff --git a/figures/beacon/Makefile b/figures/beacon/Makefile
index 7bf6c17..7af0524 100644
--- a/figures/beacon/Makefile
+++ b/figures/beacon/Makefile
@@ -113,3 +113,6 @@ antenna_setup_four.%: src/four_antenna_setup.py
 
 antenna_setup_three.%: src/four_antenna_setup.py
 	$< --no-extra $@
+
+antenna_setup_two.%: src/four_antenna_setup.py
+	$< --no-extra $@ 2
diff --git a/figures/beacon/src/four_antenna_setup.py b/figures/beacon/src/four_antenna_setup.py
index a889b0f..53509aa 100755
--- a/figures/beacon/src/four_antenna_setup.py
+++ b/figures/beacon/src/four_antenna_setup.py
@@ -140,6 +140,7 @@ if __name__ == "__main__":
 
     parser = ArgumentParser(description=__doc__)
     parser.add_argument("fname", metavar="path/to/figure[/]", nargs="?", help="Location for generated figure, will append __file__ if a directory. If not supplied, figure is shown.")
+    parser.add_argument('num_ant', help='Number of antennas to show 2, 3', default=3, type=int)
     parser.add_argument("--no-extra", dest='extra', action='store_false', help='Disable the extra (fourth) antenna')
 
     args = parser.parse_args()
@@ -155,6 +156,9 @@ if __name__ == "__main__":
         Antenna(name='3', x=1, y= 3, t0=10 ),
         ]
 
+    if args.num_ant < 3:
+        ants = [ants[0], ants[1]]
+
     if args.extra:
         extra_ant = Antenna(name='4', x=4, y=-1, t0=-6)
         all_ants = ants + [extra_ant]