Cleaned up the UI code.
Should stop some looping that was seen.

Signed-off-by: Michel <info@1st-setup.nl>