Add control buttons and wire them up