Figured out the controller concept