Update UI.c

Done button works now