Wiring up all the pieces in controller.py. Still very basic but should be functional