UI functionality restored