Add some functionally to UI