Made a new main for controller.py using argparse