More updates for UI