Adding some UI functionality.