Removed debug code from UI