Update with the newest UI changes