Moving UI drawing into own function. Adding quit to the control key.