Added missing UI features