Should move this component to shared UI