Changes I would make to the UI and why