UI functionality update